package lab.anoper.juc;

import java.util.concurrent.BrokenBarrierException;
import java.util.concurrent.CyclicBarrier;

/**
 * @author 闫家洪（yanjiahong@vcredit.com）
 * @since 2024/12/11
 */
public class CyclicBarrierExample {

    public static void main(String[] args) {
        // 创建一个CyclicBarrier，设定需要等待的线程数
        int numberOfThreads = 3;
        CyclicBarrier barrier = new CyclicBarrier(numberOfThreads, () -> {
            // 当所有线程都到达屏障时，这个任务会执行
            System.out.println("所有线程已经到达，继续执行！");
        });

        // 启动多个线程
        for (int i = 0; i < numberOfThreads; i++) {
            new Thread(new Worker(barrier, i)).start();
        }
    }

    static class Worker implements Runnable {
        private CyclicBarrier barrier;
        private int threadId;

        Worker(CyclicBarrier barrier, int threadId) {
            this.barrier = barrier;
            this.threadId = threadId;
        }

        @Override
        public void run() {
            try {
                // 模拟线程执行的任务
                System.out.println("线程 " + threadId + " 正在执行任务...");

                // 每个线程在这里等待，直到所有线程都到达屏障
                barrier.await();

                // 所有线程到达后，继续执行后续任务
                System.out.println("线程 " + threadId + " 继续执行!");
            } catch (InterruptedException | BrokenBarrierException e) {
                e.printStackTrace();
            }
        }
    }
}
